'''
This is for Figure 7.
'''

import numpy as np
import matplotlib.pyplot as plt



def draw_plot(All_KTau_list):
    color = ['hotpink', 'skyblue','darksalmon']
    plt.figure(figsize=(8, 4))
    # for i, KTau_list in enumerate(All_KTau_list):
    #     plt.plot(np.arange(1, 13, 1), KTau_list, color[i], linestyle=':', marker='^', label='{}'.format(i))
    plt.plot(np.arange(1, 13, 1), All_KTau_list[0], color[1], linestyle=':', marker='^', label='Pred')
    plt.plot(np.arange(1, 13, 1), All_KTau_list[1], color[2], linestyle=':', marker='^', label='Origin')
    plt.xticks([i for i in range(1,13,1)])
    plt.legend(loc="best")
    plt.xlabel('Month')
    plt.ylabel('Moisture10')
    plt.show()


if __name__ == '__main__':
    # train_num = 424
    # test_num = 5000
    # integers2one_hot = True
    # more_train_data = True
    # All_KTau_list = [[0.9336939255396525,0.9389022588729858, 0.9375333786010742, 0.9336939255396525, 0.9336939255396525,0.9371995131174723,0.9348624547322592,0.9423410892486572,0.93626469373703,0.9394698143005371],
    #                  [0.9357972542444865,0.9353298743565878,0.9393362800280253,0.9386685490608215,0.9414396286010742,0.9377670685450236,0.9379673997561137,0.9367320934931437,0.9383012652397156,0.9340611497561137],
    #                  [0.934461792310079, 0.9399372339248657,0.9352964560190836,0.9423410892486572,0.9399372339248657,0.9371995131174723, 0.93626469373703,0.9396701256434122,0.9364983836809794,0.9399372339248657]]
    # draw_plot(All_KTau_list)
    # All_KTau_list =[[0.038,0.19,0.19,0.18,0.20,0.283,0.352,0.38,0.40,0.331,0.311,0.162],
    #             [-0.017,-0.017,0.172,0.172,0.199,0.27,0.349,0.357,0.317,0.243,0.203,0.199]]

    # All_KTau_list = [[0.646,0.660,0.737,0.737,0.759,0.847,0.980,0.99,0.857,0.738,0.76,0.65],
    #                  [0.636,0.652,0.717,0.727,0.739,0.827,0.964,0.983,0.847,0.738,0.71,0.648]]

    All_KTau_list = [[15.57,14.01,15.27,15.28,15.77,19.66,24,23.35,23.01,16.99,14.91,13.14],
                     [15.17,13.93,14.37,14.18,14.77,17.66,22.1,20.36,21.01,16.93,13.91,13.14]]
    draw_plot(All_KTau_list)




